51. N-Queens Leetcode Python

The n-queens puzzle is the problem of placing n queens on an n×n chessboard such that no two queens attack each other.

Given an integer n, return all distinct solutions to the n-queens puzzle.

Each solution contains a distinct board configuration of the n-queens' placement, where 'Q' and '.' both indicate a queen and an empty space respectively.

For example,
There exist two distinct solutions to the 4-queens puzzle:

[
 [".Q..",  // Solution 1
  "...Q",
  "Q...",
  "..Q."],

 ["..Q.",  // Solution 2
  "Q...",
  "...Q",
  ".Q.."]

]

这题的解法和前面一题N queen II 一样,规则是两个queen不能再同一个行,列或者diagnal 所以每个循环里面有 board[i]!=j abs(k-i)!=abs[board[i]-j]

其他的是常规的dfs解法

代码如下:

class Solution:
    # @return a list of lists of string
    def solveNQueens(self, n):
        def check(k,j,board):
            for i in range(k):
                if board[i]==j or abs(k-i)==abs(board[i]-j):
                    return False
            return True
        def dfs(depth,board,valuelist,solution):
            #for i in range(len(board)):
            if depth==len(board):
                solution.append(valuelist)
            for row in range(len(board)):
                if check(depth,row,board):
                    s='.'*len(board)
                    board[depth]=row
                    dfs(depth+1,board,valuelist+[s[:row]+'Q'+s[row+1:]],solution)
        board=[-1 for i in range(n)]
        solution=[]
        dfs(0,board,[],solution)
        return solution


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
以下是使用ABT算法求解四皇后问题的Python代码: ``` def abt_search(csp): return abt_recursive(csp, {}) def abt_recursive(csp, assignment): if len(assignment) == len(csp.variables): return assignment var = select_unassigned_variable(csp, assignment) for value in order_domain_values(var, assignment, csp): if is_consistent(var, value, assignment, csp): csp.assign(var, value) inferences = {} for neighbor in csp.neighbors(var): if neighbor not in assignment: for removed_value in csp.curr_domain_values(neighbor): if not csp.constraints(var, value, neighbor, removed_value): inferences[neighbor] = removed_value csp.eliminate(neighbor, removed_value) result = abt_recursive(csp, assignment) if result is not None: return result csp.unassign(var) for neighbor, removed_value in inferences.items(): csp.revise(neighbor, removed_value) return None def select_unassigned_variable(csp, assignment): for var in csp.variables: if var not in assignment: return var def order_domain_values(var, assignment, csp): return csp.curr_domain_values(var) def is_consistent(var, value, assignment, csp): for neighbor in csp.neighbors(var): if neighbor in assignment: if not csp.constraints(var, value, neighbor, assignment[neighbor]): return False return True class CSP: def __init__(self, variables, domains, constraints): self.variables = variables self.domains = domains self.constraints = constraints self.unassigned_vars = set(variables) self.curr_domains = {v: set(domains) for v in variables} def assign(self, var, value): self.unassigned_vars.remove(var) self.curr_domains[var] = set([value]) def unassign(self, var): self.unassigned_vars.add(var) self.curr_domains[var] = set(self.domains) def eliminate(self, var, value): self.curr_domains[var].remove(value) def revise(self, var, value): self.curr_domains[var].add(value) def curr_domain_values(self, var): return self.curr_domains[var] def neighbors(self, var): return set(self.variables) - set([var]) def is_complete(self, assignment): return len(assignment) == len(self.variables) def four_queens_csp(): variables = ['Q1', 'Q2', 'Q3', 'Q4'] domains = range(1, 5) constraints = lambda Q1, Q2, Q3, Q4: (Q1 != Q2 and Q1 != Q3 and Q1 != Q4 and Q2 != Q3 and Q2 != Q4 and Q3 != Q4 and abs(Q1 - Q2) != 1 and abs(Q1 - Q3) != 2 and abs(Q1 - Q4) != 3 and abs(Q2 - Q3) != 1 and abs(Q2 - Q4) != 2 and abs(Q3 - Q4) != 1) return CSP(variables, domains, constraints) if __name__ == '__main__': csp = four_queens_csp() solution = abt_search(csp) if solution is not None: print(solution) else: print('No solution found') ``` 该代码定义了一个CSP类,其中包含“assign”、“unassign”、“eliminate”和“revise”等方法,用于实现ABT算法。它还定义了一个“four_queens_csp”函数,用于创建四皇后问题的CSP实例。最后,它调用“abt_search”函数来解决问题,并输出解决方案。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值